/*
 * This file is part of Applied Energistics 2. Copyright (c) 2013 - 2014, AlgorithmX2, All rights reserved. Applied
 * Energistics 2 is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser General
 * Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any
 * later version. Applied Energistics 2 is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
 * Public License for more details. You should have received a copy of the GNU Lesser General Public License along with
 * Applied Energistics 2. If not, see <http://www.gnu.org/licenses/lgpl>.
 */

package appeng.recipes.game;

import java.util.HashMap;
import java.util.Map;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;

import net.minecraft.inventory.IInventory;
import net.minecraft.inventory.InventoryCrafting;
import net.minecraft.item.ItemStack;
import net.minecraft.item.crafting.IRecipe;
import net.minecraft.world.World;

import com.google.common.base.Optional;

import appeng.api.AEApi;
import appeng.api.definitions.IBlocks;
import appeng.api.definitions.IDefinitions;
import appeng.api.definitions.IItemDefinition;
import appeng.api.definitions.IItems;
import appeng.api.definitions.IMaterials;
import appeng.api.storage.IMEInventory;
import appeng.api.storage.StorageChannel;
import appeng.api.storage.data.IAEItemStack;
import appeng.api.storage.data.IItemList;

public final class DisassembleRecipe implements IRecipe {

    private static final ItemStack MISMATCHED_STACK = null;

    private final Map<IItemDefinition, IItemDefinition> cellMappings;
    private final Map<IItemDefinition, IItemDefinition> nonCellMappings;

    public DisassembleRecipe() {
        final IDefinitions definitions = AEApi.instance().definitions();
        final IBlocks blocks = definitions.blocks();
        final IItems items = definitions.items();
        final IMaterials mats = definitions.materials();

        this.cellMappings = new HashMap<>(8);
        this.nonCellMappings = new HashMap<>(9);

        this.cellMappings.put(items.cell1k(), mats.cell1kPart());
        this.cellMappings.put(items.cell4k(), mats.cell4kPart());
        this.cellMappings.put(items.cell16k(), mats.cell16kPart());
        this.cellMappings.put(items.cell64k(), mats.cell64kPart());
        this.cellMappings.put(items.cell256k(), mats.cell256kPart());
        this.cellMappings.put(items.cell1024k(), mats.cell1024kPart());
        this.cellMappings.put(items.cell4096k(), mats.cell4096kPart());
        this.cellMappings.put(items.cell16384k(), mats.cell16384kPart());

        this.nonCellMappings.put(items.encodedPattern(), mats.blankPattern());
        this.nonCellMappings.put(blocks.craftingStorage1k(), mats.cell1kPart());
        this.nonCellMappings.put(blocks.craftingStorage4k(), mats.cell4kPart());
        this.nonCellMappings.put(blocks.craftingStorage16k(), mats.cell16kPart());
        this.nonCellMappings.put(blocks.craftingStorage64k(), mats.cell64kPart());
        this.nonCellMappings.put(blocks.craftingStorage256k(), mats.cell256kPart());
        this.nonCellMappings.put(blocks.craftingStorage1024k(), mats.cell1024kPart());
        this.nonCellMappings.put(blocks.craftingStorage4096k(), mats.cell4096kPart());
        this.nonCellMappings.put(blocks.craftingStorage16384k(), mats.cell16384kPart());
    }

    @Override
    public boolean matches(final InventoryCrafting inv, final World w) {
        return this.getOutput(inv) != null;
    }

    @Nullable
    private ItemStack getOutput(final IInventory inventory) {
        int itemCount = 0;
        ItemStack output = MISMATCHED_STACK;

        for (int slotIndex = 0; slotIndex < inventory.getSizeInventory(); slotIndex++) {
            final ItemStack stackInSlot = inventory.getStackInSlot(slotIndex);
            if (stackInSlot != null) {
                // needs a single input in the recipe
                itemCount++;
                if (itemCount > 1) {
                    return MISMATCHED_STACK;
                }

                // handle storage cells
                for (final ItemStack storageCellStack : this.getCellOutput(stackInSlot).asSet()) {
                    // make sure the storage cell stackInSlot empty...
                    final IMEInventory<IAEItemStack> cellInv = AEApi.instance().registries().cell()
                            .getCellInventory(stackInSlot, null, StorageChannel.ITEMS);
                    if (cellInv != null) {
                        final IItemList<IAEItemStack> list = cellInv
                                .getAvailableItems(StorageChannel.ITEMS.createList());
                        if (!list.isEmpty()) {
                            return null;
                        }
                    }

                    output = storageCellStack;
                }

                // handle crafting storage blocks
                for (final ItemStack craftingStorageStack : this.getNonCellOutput(stackInSlot).asSet()) {
                    output = craftingStorageStack;
                }
            }
        }

        return output;
    }

    @Nonnull
    private Optional<ItemStack> getCellOutput(final ItemStack compared) {
        for (final Map.Entry<IItemDefinition, IItemDefinition> entry : this.cellMappings.entrySet()) {
            if (entry.getKey().isSameAs(compared)) {
                return entry.getValue().maybeStack(1);
            }
        }

        return Optional.absent();
    }

    @Nonnull
    private Optional<ItemStack> getNonCellOutput(final ItemStack compared) {
        for (final Map.Entry<IItemDefinition, IItemDefinition> entry : this.nonCellMappings.entrySet()) {
            if (entry.getKey().isSameAs(compared)) {
                return entry.getValue().maybeStack(1);
            }
        }

        return Optional.absent();
    }

    @Nullable
    @Override
    public ItemStack getCraftingResult(final InventoryCrafting inv) {
        return this.getOutput(inv);
    }

    @Override
    public int getRecipeSize() {
        return 1;
    }

    @Nullable
    @Override
    public ItemStack getRecipeOutput() // no default output..
    {
        return null;
    }
}
